package com.soos.plugin.ds.spring.boot.autoconfigure;


public class DBHolder {

    private static final ThreadLocal<DBInfo> LOCAL = new ThreadLocal<>();

    /**
     * 获取线程本地变量中的数据库信息
     */
    public static DBInfo getDB() {
        return LOCAL.get();
    }

    /**
     * 设置线程本地变量中的数据库信息
     */
    public static void setDB(DBInfo dbInfo) {
        LOCAL.set(dbInfo);
    }

    /**
     * 移除线程本地变量中的数据库信息
     */
    public static void remove() {
        LOCAL.remove();
    }

}
